package com.example.algorithm.linear.link;

import org.w3c.dom.Node;

import java.util.Iterator;

/**
 * @program: algorithm
 * @description:
 * @packagename: com.example.algorithm.linear.link
 * @author: Jay
 * @date: 2022/04/05 23:33:04
 **/
public class LinkList<T> implements Iterable<T> {
    //记录头结点
    private Node head;
    //记录链表的长度
    private int N;



    //节点类
    private class Node {
        //存储数据
        T item;
        //下一个节点
        Node next;

        public Node(T item, Node next){
            this.item = item;
            this.next = next;
        }
    }

    public LinkList(){
        //初始化头结点
        this.head = new Node(null,null);
        //初始化元素个数
        this.N = 0;
    }

    //清空链表
    public void clear(){
        head.next = null;
        this.N = 0;
    }

    //获取链表的长度
    public int length(){
        return this.N;
    }

    //判断链表是否为空
    public boolean isEmpty(){
        return N == 0;
    }

    //获取指定位置i处的元素
    public T get(int i){
        //通过循环，从头结点开始向后找，依次找i次，就可以找到对应的元素
        Node n = head.next;
        for (int index = 0; index < i; index++){
            n = n.next;
        }

        return n.item;
    }

    //向列表中插入元素
    public void insert(T t){
        //找到当前最后一个节点

        Node n = head;
        while (n.next != null){
            n = n.next;
        }
        //创建新节点，保存元素t
        Node node = new Node(t, null);
        //让当前最后一个节点指向新节点
        n.next = node;
        //元素的个数++
        N++;
    }

    //向指定位置i处，添加一个元素
    public void insert(int i, T t){
        //找到i位置前一个节点
        Node pre = head;
        for (int index = 0; index <= i-1; index++) {
            pre = pre.next;
        }
        //找到i位置的节点
        Node curr = pre.next;
        //创建新节点，将新节点指向i位置的节点
        Node node = new Node(t, curr);
        //将i位置前一个节点指向新节点
        pre.next = node;
        //N++
        N++;
    }

    //删除i位置的节点
    public T remove(int i){
        //找到i前一个节点
        Node pre = head;
        for (int index = 0; index <= i-1; index++) {
            pre = pre.next;
        }
        //找到i位置的节点
        Node curr = pre.next;
        //找到i位置后一个的节点
        Node nextNode = curr.next;
        //将i前指向i后
        pre.next = nextNode;
        //N--
        N--;
        return curr.item;
    }

    //查找元素t在链表中第一次存在的位置
    public int indexOf(T t){
        //从头节点，依次找到每一个节点，依次和t比较,如果相同就找到了，如果没找到返回-1

        Node n = head;
        for (int i = 0; n.next != null; i++) {
            n = n.next;
            if (n.item.equals(t)){
                return i;
            }
        }

        return -1;
    }


    @Override
    public Iterator<T> iterator() {
        return new LIterator();
    }

    private class LIterator implements Iterator{
        LinkList.Node n;
        public LIterator(){
            this.n = head;
        }
        @Override
        public boolean hasNext() {
            return n.next != null;
        }

        @Override
        public Object next() {
            n = n.next;
            return n.item;
        }
    }

    /**
     * 反转整个链表
     */
    public void reverse(){
        //判断当前链表是否为空链表，如果是空，结束，如果不是则调用重载的reverse方法完成反转
        if (isEmpty()){
            return;
        }
        reverse(head.next);
    }

    //反转指定的节点，并将反转后的节点返回
    public Node reverse(Node curr){
        if (curr.next == null){
            head.next = curr;
            return curr;
        }
        //递归的反转当前curr的下一个节点，返回值是链表反转后当前结点的上一个结点
        Node pre = reverse(curr.next);
        //让返回的结点的下一个结点变成当前结点curr
        pre.next = curr;
        //把当前结点下一个结点变为null
        curr.next = null;

        return curr;
    }
}
